home *** CD-ROM | disk | FTP | other *** search
/ Night Owl 6 / Night Owl's Shareware - PDSI-006 - Night Owl Corp (1990).iso / 039a / mawk10.zip / MAIN.C < prev    next >
C/C++ Source or Header  |  1991-10-05  |  4KB  |  205 lines
  1.  
  2. /********************************************
  3. main.c
  4. copyright 1991, Michael D. Brennan
  5.  
  6. This is a source file for mawk, an implementation of
  7. the AWK programming language.
  8.  
  9. Mawk is distributed without warranty under the terms of
  10. the GNU General Public License, version 2, 1991.
  11. ********************************************/
  12.  
  13. /* $Log:    main.c,v $
  14.  * Revision 3.4.1.1  91/09/14  17:23:39  brennan
  15.  * VERSION 1.0
  16.  * 
  17.  * Revision 3.4  91/08/13  06:51:43  brennan
  18.  * VERSION .9994
  19.  * 
  20.  * Revision 3.3  91/06/29  09:47:14  brennan
  21.  * Only track NR if needed
  22.  * 
  23.  * Revision 3.2  91/06/28  04:16:58  brennan
  24.  * VERSION 0.999
  25.  * 
  26.  * Revision 3.1  91/06/08  06:14:36  brennan
  27.  * VERSION 0.995
  28.  * 
  29.  * Revision 2.6  91/06/08  06:00:18  brennan
  30.  * changed how eof is marked on main_fin
  31.  * 
  32.  * Revision 2.5  91/06/05  07:20:22  brennan
  33.  * better error messages when regular expression compiles fail
  34.  * 
  35.  * Revision 2.4  91/05/30  11:14:38  brennan
  36.  * added static 
  37.  * 
  38.  * Revision 2.3  91/05/16  12:19:56  brennan
  39.  * cleanup of machine dependencies
  40.  * 
  41.  * Revision 2.2  91/04/22  08:08:58  brennan
  42.  * cannot close(3) or close(4) because of bug in TurboC++ 1.0
  43.  * 
  44.  * Revision 2.1  91/04/08  08:23:27  brennan
  45.  * VERSION 0.97
  46.  * 
  47. */
  48.  
  49.  
  50.  
  51. /*  main.c  */
  52.  
  53. #include "mawk.h"
  54. #include "code.h"
  55. #include "init.h"
  56. #include "fin.h"
  57. #include "bi_vars.h"
  58. #include "field.h"
  59. #include "files.h"
  60. #include <stdio.h>
  61.  
  62. #if  MSDOS 
  63. void  reargv(int *, char ***) ;
  64. #endif
  65.  
  66.  
  67. static void  PROTO( execution, (void) ) ;
  68. static void  PROTO( main_loop, (void) ) ;
  69.  
  70. extern int program_fd ;
  71. char *progname ;
  72. short mawk_state ; /* 0 is compiling */
  73. short NR_reference ;
  74.  
  75. jmp_buf   exit_jump, next_jump ;
  76. int  exit_code ;
  77.  
  78.  
  79. main(argc , argv )
  80.   int argc ; char **argv ;
  81. { 
  82.  
  83. #if   MSDOS
  84.   progname = "mawk" ;
  85. #if      HAVE_REARGV
  86.   reargv(&argc, &argv) ;
  87. #endif
  88. #else    /* MSDOS */
  89. #ifdef THINK_C
  90.   progname = "MacMAWK";
  91. #else    /* THINK_C */
  92.   { char *strrchr() ;
  93.     char *p = strrchr(argv[0], '/') ;
  94.     progname = p ? p+1 : argv[0] ; }
  95. #endif
  96. #endif
  97.  
  98.   initialize(argc, argv) ;
  99.  
  100.   if ( parse() || compile_error_count )  mawk_exit(1) ;
  101.  
  102.   compile_cleanup() ;
  103.   mawk_state = EXECUTION ;
  104.   execution() ;
  105.  
  106.   mawk_exit( exit_code ) ;
  107.   return 0 ;
  108. }
  109.  
  110.  
  111. static  void  execution()
  112. {
  113.  
  114.   if ( setjmp(exit_jump) )
  115.   { if ( begin_start ) zfree(begin_start, begin_size) ;
  116.     goto the_exit ;
  117.   }
  118.  
  119.   if ( begin_start )
  120.   { (void) execute(begin_start, eval_stack-1, 0) ;
  121.     zfree( begin_start , begin_size ) ;
  122.     begin_start = (INST *) 0 ;
  123.   }
  124.  
  125.   if ( main_start || end_start )  main_loop() ;
  126.  
  127. the_exit:
  128.  
  129.   if ( setjmp(exit_jump) )  mawk_exit(exit_code) ;
  130.  
  131.   if ( main_start )  zfree(main_start, main_size) ;
  132.   if ( end_start ) (void) execute(end_start, eval_stack-1, 0) ;
  133. }
  134.  
  135.  
  136. static void  main_loop()
  137. { register char *p ;
  138.   unsigned len ;
  139.  
  140.   /* the main file stream might already be open by a call of
  141.      getline in the BEGIN block */
  142.  
  143.   if ( ! main_fin )  open_main()  ;
  144.  
  145.   if ( main_start )
  146.   {
  147.  
  148.      if ( NR_reference )
  149.      {
  150.          (void)  setjmp(next_jump) ;
  151.  
  152.      while ( p = FINgets( main_fin, &len ) )
  153.      { 
  154.        if ( TEST2(bi_vars + NR) != TWO_DOUBLES )
  155.             cast2_to_d(bi_vars + NR) ;
  156.  
  157.        bi_vars[NR].dval += 1.0 ;
  158.        bi_vars[FNR].dval += 1.0 ;
  159.  
  160.        set_field0(p, len) ;
  161.        (void) execute( main_start, eval_stack-1, 0) ;
  162.      }
  163.      }
  164.      else /* don't worry about NR */
  165.      {
  166.          (void)  setjmp(next_jump) ;
  167.  
  168.      while ( p = FINgets( main_fin, &len ) )
  169.      { 
  170.        set_field0(p, len) ;
  171.        (void) execute( main_start, eval_stack-1, 0) ;
  172.      }
  173.      }
  174.   }
  175.   else  /* eat main and set correct state */
  176.   { long nr ;
  177.     char *f0 ;
  178.     unsigned f0_len ;
  179.  
  180.     if ( TEST2(bi_vars+NR) != TWO_DOUBLES ) cast2_to_d(bi_vars+NR) ;
  181.     nr = (long) bi_vars[NR].dval ;
  182.     f0 = (char *) 0 ;
  183.  
  184.     while ( p = FINgets(main_fin, &len) )
  185.     {
  186.       f0 = p ; f0_len = len ;
  187.       nr++ ; bi_vars[FNR].dval += 1.0 ;
  188.     }
  189.     bi_vars[NR].dval = (double) nr ;
  190.     set_field0(f0, f0_len) ;
  191.   }
  192. }
  193.  
  194.  
  195. void  mawk_exit(x)
  196.   int x ;
  197. {
  198. #if  !  MSDOS
  199. #ifndef THINK_C
  200.   close_out_pipes() ;  /* no effect, if no out pipes */
  201. #endif
  202. #endif
  203.   exit(x) ;
  204. }
  205.